// my_add.cpp

#include <torch/torch.h>

torch::Tensor my_add(torch::Tensor a, torch::Tensor b)
{
    return 2 * a + b;
}
//轻松地完成了算子的实现后，我们用 PYBIND11_MODULE 来为 C++ 函数提供 Python 调用接口。
//这里的 my_lib 是我们未来要在 Python 里导入的模块名。
//双引号中的 my_add 是 Python 调用接口的名称，这里我们对齐 C++ 函数的名称，依然用 "my_add"这个名字
PYBIND11_MODULE(my_lib, m)
{
    m.def("my_add", my_add);
}
